package polarity;

import java.util.Hashtable;
import java.util.TreeMap;

public class PolarityManager
{
	PolarityPositive polarityPositive = new PolarityPositive();
	PolarityNegative polarityNegative = new PolarityNegative();
	PolarityNeutral polarityNeutral = new PolarityNeutral();
	PolarityIrrelevant polarityIrrelevant = new PolarityIrrelevant();
	
	private Hashtable<String,Polarity> polarityHt = new Hashtable<String,Polarity>();
	
	private int numberOfTweet = 0;
	
	public Hashtable<String, Polarity> getPolarityHt() { return polarityHt; }
	
	private TreeMap<String, Integer> vocabulary = new TreeMap<String, Integer>();
	
	public TreeMap<String, Integer> getVocabulary() { return vocabulary; } 
	
	public PolarityManager()
	{
		polarityHt.put("positive", polarityPositive);
		polarityHt.put("negative", polarityNegative);
		polarityHt.put("neutral", polarityNeutral);
		polarityHt.put("irrelevant", polarityIrrelevant);
	}
	
	public int getNumberOfDifferentWords()
	{
		return vocabulary.size();
	}
	
	public int getNumberOfTweet() { return numberOfTweet; }

	public void incNumberOfTweet() { numberOfTweet++; }

}
